# 调用pytorch
import torch

# torch.eye(n, m=None, out=None)
#    n (int ) – 行数
#    m (int, optional) – 列数.如果为None,则默认为n
#    out (Tensor, optinal) - Output tensor,out的张量必须提前定义，且大小必须与生成的eye矩阵大小一致


# 示例
# 创建一个3x5的单位矩阵，斜角为1，其他为0,存入a中，并在torch.eye()函数中指定了out=b
b = torch.empty(3, 5)
a = torch.eye(3, 5, out=b)
print(a)
print(b)
# tensor([[1., 0., 0., 0., 0.],
#         [0., 1., 0., 0., 0.],
#         [0., 0., 1., 0., 0.]])

